# -*- coding: utf-8 -*-
"""
-------------------------------------------------
   File Name：     model
   Description :
   Author :       Flyoung
   date：          2023/9/4
-------------------------------------------------
   Change Activity:
                   2023/9/4:
-------------------------------------------------
"""
import torch
import torch.nn as nn


class EvaluateNet(nn.Module):
    def __init__(self, learning_rate=1e-6):
        super(EvaluateNet, self).__init__()
        self.fc = nn.Sequential(
            nn.Linear(10, 32, bias=True),
            nn.ReLU(),
            nn.Linear(32, 16, bias=True),
            nn.ReLU(),
            nn.Linear(16, 1, bias=True)
        )
        self.criterion = torch.nn.MSELoss()
        self.optimizer = torch.optim.Adam(params=self.parameters(), lr=learning_rate)

    def forward(self, x):
        return self.fc(x)
